package algorithm7.leetcode.hot100.c1哈希;

import org.junit.jupiter.api.Assertions;
import org.junit.jupiter.api.Test;

import java.util.Arrays;
import java.util.HashSet;
import java.util.Set;
import java.util.stream.Collectors;

/**
 * @author merrick.peng
 */
public class h3_LC128最长连续序列 {

    public int longestConsecutive(int[] nums) {
//        return handle1(nums);
        return handle2(nums);
    }

    private int handle2(int[] nums) {
        Set<Integer> set = Arrays.stream(nums).boxed().collect(Collectors.toSet());
        int maxLen = 0;
        for (Integer num : set) {
            // 从最大的开始找【也就是说数组(1,2,3,4,5)；当num=5时才往下找最长连续序列,防止重复遍历使算法退化为n²】
            if (!set.contains(num + 1)) {
                // 当前轮次最大值
                int max_curRound = 1;
                // 遍历找
                while (set.contains(num - 1)) {
                    num--;
                    max_curRound++;
                }
                maxLen = Math.max(maxLen, max_curRound);
            }
        }
        return maxLen;
    }

    private int handle1(int[] nums) {
        // 去重
        Set<Integer> set = new HashSet<>();
        Arrays.stream(nums).forEach(set::add);

        int res = 0;
        for (Integer num : set) {
            // 从num-1开始
            if (!set.contains(num - 1)) {
                int curNum = num;
                int curRes = 1;
                while (set.contains(curNum + 1)) {
                    curNum++;
                    curRes++;
                }
                res = Math.max(curRes, res);
            }
        }
        return res;
    }

    /*
 128. 最长连续序列
给定一个未排序的整数数组 nums ，找出数字连续的最长序列（不要求序列元素在原数组中连续）的长度。
请你设计并实现时间复杂度为 O(n) 的算法解决此问题。

示例 1：
输入：nums = [100,4,200,1,3,2]
输出：4
解释：最长数字连续序列是 [1, 2, 3, 4]。它的长度为 4。

示例 2：
输入：nums = [0,3,7,2,5,8,4,6,0,1]
输出：9
     */

    @Test
    void t() {
        int[] nums = {100, 4, 200, 1, 3, 2};
        int ans = longestConsecutive(nums);
        Assertions.assertEquals(4, ans);
    }

}
